Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add aten::_ctc_loss and its variants #925

Open
wants to merge 14 commits into
base: main
Choose a base branch
from
Open

Add aten::_ctc_loss and its variants #925

wants to merge 14 commits into from

Conversation

xytintel
Copy link
Contributor

@xytintel xytintel commented Sep 19, 2024 •
edited
Loading

  • _ctc_loss
  • _ctc_loss.Tensor
  • _ctc_loss_backward
  • _ctc_loss_backward.Tensor

dvrogozh
dvrogozh suggested changes Sep 19, 2024
View reviewed changes
Copy link
Contributor

@dvrogozh dvrogozh left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ops should be excluded from here:

torch-xpu-ops/src/ATen/native/xpu/XPUFallback.template

Lines 169 to 170 in c134991

"_ctc_loss",
"_ctc_loss_backward",

and added to
https://github.com/intel/torch-xpu-ops/blob/main/yaml/xpu_functions.yaml

@dvrogozh dvrogozh mentioned this pull request Sep 19, 2024
Open
@xytintel xytintel marked this pull request as draft September 20, 2024 05:40
@xytintel xytintel added the kernel_ready dev status: kernel ready label Sep 24, 2024
@xytintel xytintel added this to the PT2.6 milestone Sep 24, 2024
@xytintel xytintel marked this pull request as ready for review September 25, 2024 02:42
@xytintel xytintel added the operator_ready dev status: operator ready label Sep 25, 2024
@xytintel xytintel marked this pull request as draft September 27, 2024 05:48
@xytintel xytintel marked this pull request as ready for review September 27, 2024 07:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@fengyuan14 fengyuan14 fengyuan14 left review comments

@dvrogozh dvrogozh dvrogozh requested changes

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
kernel_ready dev status: kernel ready operator_ready dev status: operator ready
Projects
None yet
Milestone
PT2.6
Development

Successfully merging this pull request may close these issues.
3 participants
@xytintel @dvrogozh @fengyuan14